old scripts/sim_kara_YAwell.R

library(SDEtransport)

boots = 500
type = "YMmis"
load("func_forms.RData")

if (type == "YMmis") {
  func_list = func_formsYM$func_listYMmis
  forms = func_formsYM$formsYAwell
}

if (type == "YSmis") {
  func_list = func_formsYS$func_listYSmis
  forms = func_formsYS$formsYAwell
}

if (type == "YZmis") {
  func_list = func_formsYZ$func_listYZmis
  forms = func_formsYZ$formsYAwell
}

system(paste0("mkdir -p ", paste0("results", type)))
  
# unique(data[data$M==1 &data$S ==0, c(2,4,5)])
  sim_kara = function(n, forms, truth, B = 500) {
    # truth = func_list
    # n=100
    data = gendata.SDEtransport(n, 
                                f_W = truth$f_W, 
                                f_S = truth$f_S, 
                                f_A = truth$f_A, 
                                f_Z = truth$f_Z, 
                                f_M = truth$f_M, 
                                f_Y = truth$f_Y)
    SDE_glm4(data, truth = truth,
             truncate = list(lower =.0001, upper = .9999),
             B=B, forms = forms, RCT = 0.5)
  }
  
  library(parallel)
  
  B = 1000
  n=100

  res100_YAwell = mclapply(1:B, FUN = function(x) sim_kara(n=100, forms=forms, truth=func_list, B = boots),
                           mc.cores = getOption("mc.cores", 20L))

  save(res100_YAwell, func_list, forms, file = paste0("results", type,"/res100_YAwell.RData"))

  B = 1000
  n=500

  res500_YAwell = mclapply(1:B, FUN = function(x) sim_kara(n=500, forms=forms, truth=func_list, B = boots),
                           mc.cores = getOption("mc.cores", 20L))

  save(res500_YAwell, func_list, forms, file = paste0("results", type,"/res500_YAwell.RData"))

  rm("res100_YAwell", "res500_YAwell")
  
  B = 1000
  n=1000
  
  res10000_YAwell = mclapply(1:B, FUN = function(x) sim_kara(n=5000, forms=forms, truth=func_list, B = boots), 
                            mc.cores = getOption("mc.cores", 20L))
  
  save(res10000_YAwell, func_list, forms, file = paste0("results", type,"/res10000_YAwell.RData"))
jlstiles/SDE_transport documentation built on Feb. 6, 2020, 2:06 p.m.